#pragma once

#include <unistd.h>
#include <fcntl.h>

void SetNonBlockOrDie(int sock)
{
    int f = fcntl(sock, F_GETFL);
    if (f < 0)
    {
        exit(1);
    }
    fcntl(sock, F_SETFL, f | O_NONBLOCK);
}